print matrix in python